# -*-coding: Utf-8 -*-
# @File : curd .py
# author: Chimengmeng
# blog_url : https://www.cnblogs.com/dream-ze/
# Time：2023/10/7

from sqlalchemy.orm import Session

from FastapiTest.models import models_base
from FastapiTest.schemas import schemas_base


# 查询全部数据
def get_all(db: Session):
    # 通过数据库对象 查询模型表中的 City 模型 ，过滤出 City.id == 输入的城市ID 的数据取出来
    return db.query(models_base.User).all()


# 查询指定ID对应的数据
def get_one(db: Session, user_id: int):
    # 通过数据库对象 查询模型表中的 City 模型 ，过滤出 City.id == 输入的城市ID 的数据取出来
    return db.query(models_base.User).filter(models_base.User.id == user_id).first()


# 创建城市数据
def create_user(db: Session, user: schemas_base.CreateUser):
    # 初始化城市数据对象
    print(user)
    db_user = models_base.User(**user.dict())
    # 提交数据库
    db.add(db_user)
    # 执行事务
    db.commit()
    # 刷新数据
    db.refresh(db_user)
    # 将创建好的城市对象返回
    return db_user
